Hands-On Deep Learning with Go by Gareth Seneque

Author:Gareth Seneque
Language: eng
Format: epub
Tags: COM044000 - COMPUTERS / Neural Networks, COM004000 - COMPUTERS / Intelligence (AI) and Semantics, COM042000 - COMPUTERS / Natural Language Processing
Publisher: Packt Publishing
Published: 2019-08-07T06:14:33+00:00

meanScratch := &gpuScratchOp{scratchOp{x.Shape().Clone(), dt, "mean"}}

varianceScratch := &gpuScratchOp{scratchOp{x.Shape().Clone(), dt, "variance"}}

cacheMeanScratch := &gpuScratchOp{scratchOp{scratchShape, dt, "cacheMean"}}

cacheVarianceScratch := &gpuScratchOp{scratchOp{scratchShape, dt, "cacheVariance"}}

We then create the equivalent variables in our computation graph:

g := x.Graph()

dims := len(x.Shape())

mean := G.NewTensor(g, dt, dims, G.WithShape(scratchShape.Clone()...), G.WithName(x.Name()+"_mean"), G.WithOp(meanScratch))

Download

Copyright Disclaimer:
This site does not store any files on its server. We only index and link to content provided by other sites. Please contact the content providers to delete copyright contents if any and email us, we'll remove relevant links or contents immediately.